package com.zhj.edu.slot.impl;

import com.zhj.edu.constance.RedisConstance;
import com.zhj.edu.exception.handler.BusinessException;
import com.zhj.edu.manager.RedisManager;
import com.zhj.edu.result.ResponseCode;
import com.zhj.edu.slot.generate.GenerateSlot;
import com.zhj.edu.utils.IpUtils;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.servlet.http.HttpServletRequest;

/**
 * 黑名单检测slot
 */
@Component
@Slf4j
public class BlackSlot extends GenerateSlot {

    @Autowired
    private RedisManager redisManager;

    @Override
    public int getOrder() {
        return Integer.MIN_VALUE+2;
    }

    @Override
    public boolean dealReq(HttpServletRequest req) {
        log.info("黑名单检测...");
        //这个SLOT和前面的Limit有关系  Limit满足条件之后 就会将这个数据存放到Redis
        //那么如果在Redis中存在的话 那么就会直接拒绝执行
        //首先获取ip地址
        String ipAddress = IpUtils.getIpAddress(req);
        //构建这个黑名单的key
        String blackKey = RedisConstance.getBlackPrefix()+ipAddress;
        //接下来只需要判断这个key是否存在就可以了...
        boolean exists = redisManager.existsKey(blackKey);
        if (exists){
            log.warn("当前ip地址是属于咋们的黑名单，目前无法访问，改天来试试");
            //说明这个ip地址是属于黑名单中的ip地址
            throw new BusinessException(ResponseCode.BLACK_EXCEPTION.getCode(), ResponseCode.BLACK_EXCEPTION.getMsg());
        }
        //执行到这里 说明这个ip地址不在黑名单中
        log.info("黑名单校验通过....");
        return true;
    }
}
